import torch
from torch import nn

class MLP(nn.Module):
    def __init__():
        super().__init__()
    
    def forward(x):
        return x